from typing import Dict, Any
from .faiss_service import FAISSService, INDEX_NAME


class LocalRetriever:
	"""本地轻量检索器：使用FAISSService检索，演示用。
	返回是否有命中，用于驱动“有候选商品”的分支。
	"""

	def __init__(self, index_name: str = "agent_" + INDEX_NAME):
		self.faiss = FAISSService(index_name)

	def query(self, text: str) -> Dict[str, Any]:
		# 这里不做复杂检索，判断向量库是否加载成功即可
		try:
			_ = self.faiss.vectorstore
			return {"ok": True, "hits": len(self.faiss.documents)}
		except Exception:
			return {"ok": False, "hits": 0}


